package com.heima.wemedia.service.impl;

import com.alibaba.fastjson.JSON;
import com.heima.aliyun.GreenImageScan;
import com.heima.aliyun.GreenTextScan;
import com.heima.common.constants.message.NewsAutoScanConstants;
import com.heima.common.constants.message.PublishArticleConstants;
import com.heima.common.exception.CustException;
import com.heima.feigns.AdminFeign;
import com.heima.model.common.dtos.ResponseResult;
import com.heima.model.common.enums.AppHttpCodeEnum;
import com.heima.model.wemedia.pojos.WmNews;
import com.heima.utils.common.SensitiveWordUtil;
import com.heima.wemedia.mapper.WmNewsMapper;
import com.heima.wemedia.service.WmNewsAutoScanService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.amqp.AmqpException;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.core.MessagePostProcessor;
import org.springframework.amqp.core.MessageProperties;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import java.util.*;
import java.util.stream.Collectors;
import java.util.stream.Stream;


@Service
@Slf4j
public class WmNewsAutoScanServiceImpl implements WmNewsAutoScanService {


    /**
     * 注入adminfeign
     * @param id
     */
    @Autowired
    private AdminFeign adminFeign;

    @Autowired
    private WmNewsMapper wmNewsMapper;

    @Autowired
    private GreenImageScan greenImageScan;

    @Autowired
    private GreenTextScan greenTextScan;


    @Autowired
    private RabbitTemplate rabbitTemplate;



    /**
     * 注入图片前缀
     */
    @Value("${file.oss.web-site}")
    private String webSite;


    /**
     * 实现审核文章功能
     * @param wmNewsId  自媒体文章id
     */
    @Override
    public void autoScanWmNews(Integer wmNewsId) {
        log.info(" 自动审核发布方法 被调用   当前审核发布的文章id==> {}",wmNewsId);
        //验证参数
        if (wmNewsId == null){
            log.error("自动审核文章失败    文章id为空");
            CustException.cust(AppHttpCodeEnum.PARAM_REQUIRE,"数据有误");
        }
        //根据id查询对应的文章
        WmNews wmNews = wmNewsMapper.selectById(wmNewsId);
        //判断数据
        if (wmNews == null){
            log.error("自动审核文章失败    未查询自媒体文章信息  wmNewsId:{}",wmNewsId);
            CustException.cust(AppHttpCodeEnum.DATA_NOT_EXIST,"对应的文章不存在");
        }
        //判断文章的状态
        if (wmNews.getStatus() != WmNews.Status.SUBMIT.getCode()){
            CustException.cust(AppHttpCodeEnum.DATA_NOT_ALLOW,"数据不在审核状态");
        }

        // 抽取出文章中 所有的文本内容 和 所有的图片url集合    Map<String,Object>  content 内容   images List<String>
        //将文章中 文本的内容抽取出来 检查文本敏感词
        //将文章中的图片抽取出来  检查图片敏感
        Map<String,Object> content = handleTextAndImages(wmNews);
        //获取文本和图片内容
        List<String> image = (List<String>)content.get("image");
        String text = (String) content.get("text");

        // 3.1  敏感词审核    失败   修改文章状态(2)
        boolean isSensivice = handleSensitive(text,wmNews);
        if (!isSensivice){
            return;
        }

        // 3.2  阿里云的文本审核   失败  状态2  不确定 状态3
        boolean isTextScan = handleTextScan(text,wmNews);
        if (!isTextScan){
            return;
        }

        // 3.3  阿里云的图片审核   失败  状态2  不确定 状态3
        boolean isImageScan = handleImageScan(image , wmNews);
        if (!isImageScan){
            return;
        }

        // 4. 判断文章发布时间是否大于当前时间   状态 8
        updateWmNews(wmNews,WmNews.Status.SUCCESS.getCode(),"审核通过");
        log.info("审核通过-----------------------------------------");


        // todo 5. 通知定时发布文章
        //文章准备就绪   将文章发送到队列   准备发布
//        //发布时间
//        long timeOut = wmNews.getPublishTime().getTime();
//        //当前时间
//        long timeNow = new Date().getTime();
//        //时间差
//        long time = timeOut - timeNow;

        //发布文章到交换机
//        rabbitTemplate.convertAndSend(PublishArticleConstants.DELAY_DIRECT_EXCHANGE,  //交换价
//                                        PublishArticleConstants.PUBLISH_ARTICLE_ROUTE_KEY,   //路由key
//                                        wmNews.getId(),   //发送的内容
//                                        (message -> {     //延迟队列的时间
//                                            message.getMessageProperties().setHeader("x-delay",time<= 0 ? 0 : time);
//                                            return message;
//                                        }));
        long publishTime = wmNews.getPublishTime().getTime();
        // 当前时间
        long nowTime = new Date().getTime();
        long remainTime = publishTime - nowTime;
        // 发布文章
        rabbitTemplate.convertAndSend(PublishArticleConstants.DELAY_DIRECT_EXCHANGE
                , PublishArticleConstants.PUBLISH_ARTICLE_ROUTE_KEY
                , wmNews.getId()
                , (message)->{                              // 延时消息 必设置
                    message.getMessageProperties().setHeader("x-delay",remainTime<=0?0:remainTime);
                    return message;
                }
        );
        log.info("立即发布文章通知成功发送，文章id : {}", wmNews.getId());


    }



    /**
     * 将文章对象中的文本内容和图片都抽取出来
     * @param wmNews
     * @return
     */
    private Map<String, Object> handleTextAndImages(WmNews wmNews) {
        //解析正文部分
        String content = wmNews.getContent();
        List<Map> maps = JSON.parseArray(content, Map.class);
        //先获取正文中的图片
        List<String> imageList = new ArrayList<>();
        imageList = maps.stream().filter(map -> map.get("type").equals("image"))
                .map(map -> map.get("value").toString())
                .collect(Collectors.toList());


        //获取封面图片
        if (StringUtils.isNotBlank(wmNews.getImages())) {
            List<String> urls = Arrays.stream(wmNews.getImages().split(","))
                    .map(url -> webSite + url)
                    .collect(Collectors.toList());
            imageList.addAll(urls);
        }
        //图片去重
        imageList = imageList.stream().distinct().collect(Collectors.toList());



        //获取正文中的文本
        String text = maps.stream().filter(map -> map.get("type").equals("text"))
                .map(map -> map.get("value").toString())
                .distinct()
                .collect(Collectors.joining("-666-"));

        //获取文章头部信息 如果有头部信息 就和正文文本拼接
        String title = wmNews.getTitle();
        if (StringUtils.isNotBlank(title)){
            text += "-666-" + title;
        }

        Map<String , Object> map = new HashMap<>();
        map.put("image" , imageList);
        map.put("text" , text);

        return map;
    }


    /**
     * 阿里云审核图片
     * @param image
     * @param wmNews
     * @return
     */
    private boolean handleImageScan(List<String> image, WmNews wmNews) {
        boolean flag = true;
        //调用阿里云审核图片
        try {
            Map imageScan = greenImageScan.imageUrlScan(image);
            String suggestion = (String) imageScan.get("suggestion");
            switch (suggestion){
                //不通过
                case "block" :
                    updateWmNews(wmNews,WmNews.Status.FAIL.getCode(),"图片中有违规内容，审核失败");
                    flag = false;
                    break;
                //需要人工审核
                case "review":
                    updateWmNews(wmNews,WmNews.Status.ADMIN_AUTH.getCode(),"图片中有不确定内容，转为人工审核");
                    flag = false;
                    break;
                default:
            }
        } catch (Exception e) {
            e.printStackTrace();
            e.printStackTrace();
            log.error("阿里云图片审核出现异常 , 原因:{}",e.getMessage());
            updateWmNews(wmNews,WmNews.Status.ADMIN_AUTH.getCode(),"阿里云内容服务异常，转为人工审核");
            flag = false;
        }
        return flag;
    }



    /**
     * 阿里云审核文本敏感词
     * @param text
     * @param wmNews
     * @return
     */
    private boolean handleTextScan(String text, WmNews wmNews) {
        boolean flag = true;
        //调用阿里云审核文本
        try {
            Map textScan = greenTextScan.greenTextScan(text);
            String suggestion = (String) textScan.get("suggestion");
            switch (suggestion){
                //不通过
                case "block" :
                    updateWmNews(wmNews,WmNews.Status.FAIL.getCode(),"文本中有违规内容，审核失败");
                    flag = false;
                    break;
                //需要人工审核
                case "review":
                    updateWmNews(wmNews,WmNews.Status.ADMIN_AUTH.getCode(),"文本中有不确定内容，转为人工审核");
                    flag = false;
                    break;
                default:
            }
        } catch (Exception e) {
            e.printStackTrace();
            e.printStackTrace();
            log.error("阿里云文本审核出现异常 , 原因:{}",e.getMessage());
            updateWmNews(wmNews,WmNews.Status.ADMIN_AUTH.getCode(),"阿里云内容服务异常，转为人工审核");
            flag = false;
        }
        return flag;
    }




    /**
     * 自定义的方法  查询文章敏感词
     * @param text
     * @param wmNews
     * @return
     */
    private boolean handleSensitive(String text, WmNews wmNews) {
        //调用feign接口远程调用  获取自定义敏感词
        ResponseResult<List<String>> responseResult = adminFeign.selectAllSensitives();
        if (responseResult.getCode() != 0){
            CustException.cust(AppHttpCodeEnum.SERVER_ERROR,"远程调用feign出错");
        }
        //获取数据
        List<String> list = responseResult.getData();
        // 2. 将敏感词集合转发DFA数据模型
        SensitiveWordUtil.initMap(list);

        //检验敏感词
        Map<String, Integer> stringIntegerMap = SensitiveWordUtil.matchWords(text);
        if (stringIntegerMap != null && stringIntegerMap.size()>0){
            //表示审核没通过
            updateWmNews(wmNews , WmNews.Status.FAIL.getCode() , "内容包含敏感词" + responseResult);
            return false;
        }
        return true;
    }


    /**
     * 修改审核状态
     * @param wmNews
     * @param status
     * @param reason
     */
    private void updateWmNews(WmNews wmNews, short status, String reason) {
        //修改状态
        wmNews.setStatus(status);
        //状态说明
        wmNews.setReason(reason);
        //修改文章状态
        wmNewsMapper.updateById(wmNews);
    }


}
